package com.webserver.http;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.Socket;

/**
 * 响应对象
 * 该类的每一个实例用于表示一个HTTP的响应内容
 * 每个响应由三部分构成:
 * 状态行,响应头,相应正文
 */

public class HttpResponse
{
    //状态行相关信息
    private String protocol = "HTTP/1.1";//协议版本
    private int statusCode = 200;//状态代码
    private String statusReason = "OK";//状态描述

    //响应头相关信息
    private File entity;//正文对应的一个实体文件
    //相应正文相关信息

    private Socket socket;

    public HttpResponse(Socket socket)
    {
        this.socket = socket;
    }

    /**
     * 将当前响应对象内容按照标准的HTTP相应格式发送给客户端
     */
    public void flush() throws IOException
    {
        //3.1:发送状态行
        sendStatusLine();
        //3.2:发送响应头
        sendHeaders();
        //3.3：发送正文（index页面内容）
        sendContent();
    }

    private void sendStatusLine() throws IOException
    {
        String line = protocol + " " + statusCode + " " + statusReason;
        println(line);
    }

    private void sendHeaders() throws IOException
    {
        String line = "Content-Type：text/html";
        println(line);

        line = "Content-Length: " + entity.length();
        println(line);

        //单独发送CRLF表示响应头发送完毕
        println("");
    }

    private void sendContent() throws IOException
    {
        OutputStream out = socket.getOutputStream();
        try (
                FileInputStream fis = new FileInputStream(entity))
        {
            byte[] buf = new byte[1028 * 8];
            int len;
            while ((len = fis.read(buf)) != -1)
            {
                out.write(buf, 0, len);
            }
        }
    }

    private void println(String line) throws IOException
    {
        OutputStream out = socket.getOutputStream();
        byte[] data = line.getBytes("iso8859-1");
        out.write(data);
        out.write(13);
        out.write(10);
    }

    public String getProtocol()
    {
        return protocol;
    }

    public void setProtocol(String protocol)
    {
        this.protocol = protocol;
    }

    public int getStatusCode()
    {
        return statusCode;
    }

    public void setStatusCode(int statusCode)
    {
        this.statusCode = statusCode;
    }

    public String getStatusReason()
    {
        return statusReason;
    }

    public void setStatusReason(String statusReason)
    {
        this.statusReason = statusReason;
    }

    public File getEntity()
    {
        return entity;
    }

    public void setEntity(File entity)
    {
        this.entity = entity;
    }
}
